python -u ./main.py \
  --device_target GPU \
  --device_id 0 \
  --output_dir ./summary